#!/usr/bin/env python3
"""
数据库重置脚本
强制删除并重新创建所有数据库表
"""

import os
from database import reset_database, get_db
# 导入所有模型确保它们被注册到 Base.metadata
from db_models import UserDB, TodoDB
from auth import init_default_users

def reset_database_completely():
    """完全重置数据库"""
    print("🔄 开始重置数据库...")
    
    # 1. 删除数据库文件
    db_file = "todos.db"
    if os.path.exists(db_file):
        os.remove(db_file)
        print(f"🗑️ 已删除数据库文件: {db_file}")
    
    # 2. 重新创建数据库表
    reset_database()
    
    # 3. 初始化默认用户（包括管理员）
    db = next(get_db())
    try:
        init_default_users(db)
        print("✅ 数据库重置完成！")
        print("👑 管理员账号: admin / admin123")
        print("👤 普通用户: zgy / 0110, hjb / 0926")
    finally:
        db.close()

if __name__ == "__main__":
    reset_database_completely()